package freedom.easyit.web;

import java.io.IOException;
import java.net.URLEncoder;
import java.text.ParseException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Locale;
import java.util.Map;

import javax.servlet.http.HttpServletResponse;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.format.datetime.DateFormatter;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.multipart.MultipartFile;

import com.alibaba.excel.EasyExcel;
import com.alibaba.fastjson.JSON;

import freedom.easyit.domain.ContactBook;
import freedom.easyit.domain.MailHistory;
import freedom.easyit.dto.ContactBookDto;
import freedom.easyit.listener.ContactBookDtoListener;
import freedom.easyit.service.MailService;

/**
 * 邮件
 * 
 * @author Charlie Email:tingfangcao@163.com
 * @date 2020/10/20
 */
@Controller
@RequestMapping("mail")
public class MailController {

    @Autowired
    private ContactBookDtoListener contactBookDtoListener;
    @Autowired
    private MailService mailService;

    /**
     * 文件下载（失败了会返回一个有部分数据的Excel）
     * <p>
     * 1. 创建excel对应的实体对象 参照{@link ContactBookDto}
     * <p>
     * 2. 设置返回的 参数
     * <p>
     * 3. 直接写，这里注意，finish的时候会自动关闭OutputStream,当然你外面再关闭流问题不大
     * 
     * @param response
     * @throws IOException
     * @throws ParseException
     */
    @GetMapping("download")
    public void download(HttpServletResponse response) throws IOException, ParseException {
        // 这里注意 有同学反应使用swagger 会导致各种问题，请直接用浏览器或者用postman
        response.setContentType("application/vnd.ms-excel");
        response.setCharacterEncoding("utf-8");
        // 这里URLEncoder.encode可以防止中文乱码 当然和easyexcel没有关系
        String fileName = URLEncoder.encode("诺祺科技", "UTF-8").replaceAll("\\+", "%20");
        response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + fileName + ".xlsx");
        EasyExcel.write(response.getOutputStream(), ContactBookDto.class).sheet("通讯录").doWrite(data());
    }

    /**
     * 文件下载并且失败的时候返回json（默认失败了会返回一个有部分数据的Excel）
     *
     * @since 2.1.1
     */
    @GetMapping("downloadFailedUsingJson")
    public void downloadFailedUsingJson(HttpServletResponse response) throws IOException {
        // 这里注意 有同学反应使用swagger 会导致各种问题，请直接用浏览器或者用postman
        try {
            response.setContentType("application/vnd.ms-excel");
            response.setCharacterEncoding("utf-8");
            // 这里URLEncoder.encode可以防止中文乱码 当然和easyexcel没有关系
            String fileName = URLEncoder.encode("测试", "UTF-8").replaceAll("\\+", "%20");
            response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + fileName + ".xlsx");
            // 这里需要设置不关闭流
            EasyExcel.write(response.getOutputStream(), ContactBookDto.class).autoCloseStream(Boolean.FALSE).sheet("模板")
                .doWrite(data());
        } catch (Exception e) {
            // 重置response
            response.reset();
            response.setContentType("application/json");
            response.setCharacterEncoding("utf-8");
            Map<String, String> map = new HashMap<String, String>();
            map.put("status", "failure");
            map.put("message", "下载文件失败" + e.getMessage());
            response.getWriter().println(JSON.toJSONString(map));
        }
    }

    private List<ContactBookDto> data() throws ParseException {
        DateFormatter dateFormatter = new DateFormatter("yyyy/MM/dd");
        List<ContactBookDto> list = new ArrayList<ContactBookDto>();
        for (int i = 0; i < 10; i++) {
            ContactBookDto data = new ContactBookDto();
            data.setName("龚成伟");
            data.setEmail("gongcw@skyon.com.cn");
            data.setHiredate(dateFormatter.parse("2020/9/10", Locale.getDefault()));
            data.setGendar("男");
            data.setBirthday(dateFormatter.parse("1987/1/1", Locale.getDefault()));
            list.add(data);
        }
        return list;
    }

    /**
     * 文件上传
     * <p>
     * 1. 创建excel对应的实体对象 参照{@link ContactBookDto}
     * <p>
     * 2. 由于默认一行行的读取excel，所以需要创建excel一行一行的回调监听器，参照{@link ContactBookDtoListener}
     * <p>
     * 3. 直接读即可
     */
    @PostMapping("upload")
    @ResponseBody
    public String upload(MultipartFile file) throws IOException {
        EasyExcel.read(file.getInputStream(), ContactBookDto.class, contactBookDtoListener).sheet().doRead();
        return "success";
    }
    
    @GetMapping("contactBook")
    @ResponseBody
    public List<ContactBook> contactBook() {
        List<ContactBook> list = mailService.listContactBook();
        return list;
    }
    @GetMapping("mailHistory10")
    @ResponseBody
    public List<MailHistory> mailHistory10() {
        List<MailHistory> list = mailService.listMailHistory10();
        return list;
    }

}
